package gitbucket.core.model

trait ProtectedBranchComponent extends TemplateComponent { self: Profile =>
  import profile.api._
  import self._

  lazy val ProtectedBranches = TableQuery[ProtectedBranches]
  class ProtectedBranches(tag: Tag) extends Table[ProtectedBranch](tag, "PROTECTED_BRANCH") with BranchTemplate {
    val statusCheckAdmin = column[Boolean]("STATUS_CHECK_ADMIN")
    def * = (userName, repositoryName, branch, statusCheckAdmin).mapTo[ProtectedBranch]
    def byPrimaryKey(userName: String, repositoryName: String, branch: String) =
      byBranch(userName, repositoryName, branch)
    def byPrimaryKey(userName: Rep[String], repositoryName: Rep[String], branch: Rep[String]) =
      byBranch(userName, repositoryName, branch)
  }

  lazy val ProtectedBranchContexts = TableQuery[ProtectedBranchContexts]
  class ProtectedBranchContexts(tag: Tag)
      extends Table[ProtectedBranchContext](tag, "PROTECTED_BRANCH_REQUIRE_CONTEXT")
      with BranchTemplate {
    val context = column[String]("CONTEXT")
    def * =
      (userName, repositoryName, branch, context).mapTo[ProtectedBranchContext]
  }
}

case class ProtectedBranch(userName: String, repositoryName: String, branch: String, statusCheckAdmin: Boolean)

case class ProtectedBranchContext(userName: String, repositoryName: String, branch: String, context: String)
